/*
 * Copyright (c) 2020 - present, Inspur Genersoft Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.inspur.edp.cef.generator.apiGenerator.dataType.entityInterface.assoInfo;

import com.inspur.edp.caf.generator.BaseGenerator;
import com.inspur.edp.caf.generator.baseInfo.TypeInfo;
import com.inspur.edp.caf.generator.item.ClassGenerator;
import com.inspur.edp.caf.generator.item.ClassGeneratorContext;
import com.inspur.edp.caf.generator.property.ClassPropertyGenerator;
import com.inspur.edp.caf.generator.property.PropertyGenerator;
import com.inspur.edp.cef.designtime.api.IGspCommonField;
import com.inspur.edp.cef.designtime.api.element.GspAssociation;
import com.inspur.edp.cef.gencommon.MediateType;
import com.inspur.edp.cef.generator.apiGenerator.dataType.entityInterface.propertyInfo.DataTypPropInfoGenCtx;
import com.inspur.edp.cef.spi.entity.AssociationInfo;
import java.util.List;
import java.util.stream.Collectors;
import com.inspur.edp.udt.designtime.api.entity.UnifiedDataTypeDef;
import org.eclipse.jdt.core.dom.Modifier.ModifierKeyword;

public class AssociationInfoGen extends ClassGenerator {
  protected DataTypPropInfoGenCtx parentContext;
  protected IGspCommonField field;
  protected GspAssociation asso;

  public AssociationInfoGen(DataTypPropInfoGenCtx parentContext, GspAssociation asso) {
    super(
        parentContext
            .getDataTypeInfoGeneratorCtx()
            .getDataTypeInterfaceCtx()
            .getApiGenContext()
            .getCompilationUnitInfo());
    this.parentContext = parentContext;
    this.field = parentContext.getCommonField();
    this.asso = asso;
  }

  @Override
  protected void beforeInitialize() {
    super.beforeInitialize();
    parentContext.setAssociationInfoType(new MediateType(super.compileUnit.getPackageName(), getName()));
  }


  // keyword:
  // ORIGINAL LINE: protected new AssociaitionInfoGenCtx getContext() { return
  // (AssociaitionInfoGenCtx)((super.getContext() instanceof AssociaitionInfoGenCtx) ?
  // super.getContext() : null); }
  @Override
  protected AssociaitionInfoGenCtx getContext() {
    return (AssociaitionInfoGenCtx)
        ((super.getContext() instanceof AssociaitionInfoGenCtx) ? super.getContext() : null);
  }

  @Override
  protected TypeInfo setParentClass() {
    return new TypeInfo(AssociationInfo.class);
  }

  @Override
  protected String getName() {
    if(!(field.getBelongObject() instanceof UnifiedDataTypeDef)){
      return field.getBelongObject().getCode() + field.getLabelID() + "AssoInfo";
    }
    else{
      return field.getLabelID() + "AssoInfo";
    }
  }

  @Override
  protected ClassGeneratorContext createClassInfoContext() {
    return new AssociaitionInfoGenCtx(parentContext, asso);
  }

  @Override
  protected java.util.ArrayList<ModifierKeyword> getAccessModifier() {
    return new java.util.ArrayList<ModifierKeyword>(
        java.util.Arrays.asList(new ModifierKeyword[] {ModifierKeyword.PUBLIC_KEYWORD}));
  }

  @Override
  protected java.util.ArrayList<ClassPropertyGenerator> createPropertyGenerators() {
    java.util.ArrayList<ClassPropertyGenerator> list = new java.util.ArrayList<ClassPropertyGenerator>();
    list.add(new AssoRefPropTypesDicGen(getContext()));
    return list;
  }

  @Override
  protected java.util.ArrayList<BaseGenerator> createClassExtendChildGenrators() {
    java.util.ArrayList<BaseGenerator> list = new java.util.ArrayList<BaseGenerator>();
    // 字段


    List<BaseGenerator> eleGenList =
        asso.getRefElementCollection().stream()
            .map(el -> (BaseGenerator) createPropInfoGenerator(el))
            .collect(Collectors.toList());
    if (eleGenList != null && eleGenList.size() > 0) {
      list.addAll(eleGenList);
    }
    return list;
  }

  protected AbstractRefPropInfoGen createPropInfoGenerator(IGspCommonField field) {
    return new AbstractRefPropInfoGen(getContext(), field);
  }
}
